import pytest

from demo import demo


class TestDemo:
    # in表示输入, out表示输出, 这样下次回来看就知道, 输入"猴", 可以输出"孙悟空", 可读性较好
    l = [
        {"in": "猴", "out": "孙悟空"},
        {"in": "猪", "out": "猪八戒"},
        {"in": "马", "out": "小白龙"}
    ]

    @pytest.mark.parametrize("data", l)
    def test_a(self, data):
        # 输出: ['猴', '孙悟空'] 猴 孙悟空, data["in"] 对应 "猴", data["out"] 对应 "孙悟空"
        print(data, data["in"], data["out"])
        # 那么断言就好写了
        assert demo(data["in"]) == data["out"]

    # @pytest.mark.parametrize("data", [["猴","孙悟空"],["猪","猪八戒"],["马","小白龙"]])
    # def test_a(self,data):
    #     # 输出: ['猴', '孙悟空'] 猴 孙悟空, data[0] 对应 "猴", data[1] 对应 "孙悟空"
    #     print(data,data[0],data[1])
    #     # 那么断言就好写了
    #     assert demo(data[0]) == data[1]
